package com.leaves.core.util;

import com.leaves.core.consts.MediaTypeConstant;

/**
 * @Author Chenweiwei
 * @Date 2022/3/22 23:00
 * @Version 1.0
 */
public class FileUtils {

    /**
     * 获取文件类型
     * 文件类型：0:文档，1图片，2音频，3视频,4其他
     *
     * @param fileSuffix
     * @return
     */
    public static Integer getFileType(String fileSuffix) {
        if (MediaTypeConstant.FILE_EXTENSION.contains(fileSuffix)) {
            return 0;
        }
        if (MediaTypeConstant.IMAGE_EXTENSION.contains(fileSuffix)) {
            return 1;
        }
        if (MediaTypeConstant.AUDIO_EXTENSION.contains(fileSuffix)) {
            return 2;
        }
        if (MediaTypeConstant.VIDEO_EXTENSION.contains(fileSuffix)) {
            return 3;
        }
        return 4;
    }
}
